AWS Step FunctionsでX-Rayを有効にしてみた（AWS CDK v2）

若槻龍太

2022.07.15

この記事は公開されてから1年以上経過しています。情報が古い可能性がありますので、ご注意ください。

こんにちは、CX事業本部 IoT事業部の若槻です。

AWS Step Functionsでは、X-Rayを有効にすることによりState Machine実行を可視化して、ボトルネックの発見やトラブルシュートに役立てることができます。

AWS X-Ray and Step Functions - AWS Step Functions

今回は、AWS CDK v2でAWS Step FunctionsのX-Rayを有効にし、動作を確認してみました。

やってみた

環境作成

AWS CDK v2（TypeScript）で次のようなCDKスタックを作成します。

import { Construct } from 'constructs';
import {
  aws_lambda_nodejs,
  aws_stepfunctions,
  aws_stepfunctions_tasks,
  Duration,
  Stack,
  StackProps,
} from 'aws-cdk-lib';
import { WaitTime } from 'aws-cdk-lib/aws-stepfunctions';

export class AwsAppStack extends Stack {
  constructor(scope: Construct, id: string, props: StackProps) {
    super(scope, id, props);

    const funcTask = new aws_stepfunctions_tasks.LambdaInvoke(
      this,
      'funcTask',
      {
        lambdaFunction: new aws_lambda_nodejs.NodejsFunction(this, 'func', {
          entry: 'src/handler.ts',
          handler: 'handler',
        }),
      },
    );

    // StateMachine(Standard)
    new aws_stepfunctions.StateMachine(this, 'stateMachine', {
      stateMachineName: 'stateMachine',
      definition: funcTask.next(
        new aws_stepfunctions.Wait(this, 'wait', {
          time: WaitTime.duration(Duration.seconds(3)),
        }),
      ),
      tracingEnabled: true,
    });

    // StateMachine(Express)
    new aws_stepfunctions.StateMachine(this, 'express', {
      stateMachineName: 'express',
      stateMachineType: aws_stepfunctions.StateMachineType.EXPRESS,
      definition: new aws_stepfunctions.Pass(this, 'pass'),
      tracingEnabled: true,
    });
  }
}

StateMachine ClassでtracingEnabledをtrueにすることにより、StateMachineのX-Rayを有効にすることができます。
StandartおよびExpressの2種類のWorkflow TypeのState Machineをそれぞれ作成して試してみます。